﻿<template>
  <div class="app-container">
    <div class="filter-container">
      <el-input
        v-model="listQuery.title"
        :placeholder="$t('table.title')"
        style="width: 200px;"
        class="filter-item"
        @keyup.enter.native="handleFilter"
      />
      <!-- <el-select v-model="listQuery.importance" :placeholder="$t('table.importance')" clearable style="width: 90px" class="filter-item">
        <el-option v-for="item in importanceOptions" :key="item" :label="item" :value="item" />
      </el-select> -->
      <el-select
        v-model="listQuery.sort"
        style="width: 140px; margin-left:20px"
        class="filter-item"
        @change="handleFilter"
      >
        <el-option v-for="item in sortOptions" :key="item.key" :label="item.label" :value="item.key" />
      </el-select>
      <el-button
        v-waves
        class="filter-item"
        type="primary"
        icon="el-icon-search"
        style="margin-left:15px"
        @click="handleFilter"
      >
        {{ $t('table.search') }}
      </el-button>
      <!-- <el-button class="filter-item" style="margin-left: 10px;" type="primary" icon="el-icon-edit" @click="handleCreate">
        {{ $t('table.add') }}
      </el-button> -->
      <!-- <el-button v-waves :loading="downloadLoading" class="filter-item" type="primary" icon="el-icon-download" @click="handleDownload">
        {{ $t('table.export') }}
      </el-button> -->
      <!-- <el-checkbox v-model="showReviewer" class="filter-item" style="margin-left:15px;" @change="tableKey=tableKey+1">
        {{ $t('table.reviewer') }}
      </el-checkbox> -->
    </div>

    <el-table
      :key="tableKey"
      v-loading="listLoading"
      :data="list"
      border
      fit
      highlight-current-row
      style="width: 100%;"
      @sort-change="sortChange"
    >
      <el-table-column
        :label="$t('table.id')"
        prop="id"
        sortable="custom"
        align="center"
        width="80"
        :class-name="getSortClass('id')"
      >
        <template slot-scope="{row}">
          <span>{{ row.id }}</span>
        </template>
      </el-table-column>
      <el-table-column label="实例名称" width="200px" align="center">
        <template slot-scope="{row}">
          <span>{{ row.InsName }}</span>
        </template>
      </el-table-column>
      <el-table-column label="创建时间" width="200px" align="center">
        <template slot-scope="{row}">
          <span>{{ row.createdTime }}</span>
        </template>
      </el-table-column>
      <el-table-column label="模型" min-width="300px">
        <template slot-scope="{row}">
          <span>{{ row.model }}</span>
          <!-- <span class="link-type" @click="handleUpdate(row)">{{ row.description }}</span> -->
          <!-- <el-tag>{{ row.type | typeFilter }}</el-tag> -->
        </template>
      </el-table-column>
      <el-table-column label="超参数" width="200px" align="center">
        <!-- <template slot-scope="{row}">
          <span>{{ row.hyperParameter }}</span>
        </template> -->
        <template slot-scope="{row,$index}">
          <el-button type="primary" size="mini" @click="handleShow(row, $index)">
            查看
          </el-button>
        </template>
      </el-table-column>
      <el-table-column label="关联数据集" min-width="110px" align="center">
        <template slot-scope="{row}">
          <span class="link-type" @click="handleUpdate(row)">{{ row.dataset }}</span>
        </template>
      </el-table-column>
      <el-table-column label="操作人员" width="110px" align="center">
        <template slot-scope="{row}">
          <span>{{ row.operator }}</span>
        </template>
      </el-table-column>
      <!-- <el-table-column v-if="showReviewer" :label="$t('table.reviewer')" width="110px" align="center">
        <template slot-scope="{row}">
          <span style="color:red;">{{ row.reviewer }}</span>
        </template>
      </el-table-column> -->
      <!-- <el-table-column :label="$t('table.importance')" width="80px">
        <template slot-scope="{row}">
          <svg-icon v-for="n in +row.importance" :key="n" icon-class="star" class="meta-item__icon" />
        </template>
      </el-table-column>
      <el-table-column :label="$t('table.readings')" align="center" width="95">
        <template slot-scope="{row}">
          <span v-if="row.pageviews" class="link-type" @click="handleFetchPv(row.pageviews)">{{ row.pageviews }}</span>
          <span v-else>0</span>
        </template>
      </el-table-column> -->
      <el-table-column label="模型状态" class-name="status-col" width="100">
        <template slot-scope="{row}">
          <el-tag :type="row.status | statusFilter">
            {{ row.status }}
          </el-tag>
        </template>
      </el-table-column>
      <el-table-column :label="$t('table.actions')" align="center" width="300" class-name="small-padding fixed-width">
        <template slot-scope="{row,$index}">
          <el-button type="primary" size="mini" @click="handleShowRes(row)">
            查看
          </el-button>
          <!-- <el-button v-if="row.status!='draft'" size="mini" @click="handleModifyStatus(row,'draft')">
            {{ $t('table.draft') }}
          </el-button> -->
          <el-button v-if="row.status!='Deleted'" size="mini" type="danger" @click="handleDelete(row,$index)">
            {{ $t('table.delete') }}
          </el-button>
        </template>
      </el-table-column>
    </el-table>

    <pagination
      v-show="total>0"
      :total="total"
      :page.sync="listQuery.page"
      :limit.sync="listQuery.limit"
      @pagination="getList"
    />

    <el-dialog :title="textMap[dialogStatus]" :visible.sync="dialogFormVisible">
      <el-form
        ref="dataForm"
        :rules="rules"
        :model="temp"
        label-position="left"
        label-width="150px"
        style="width: 400px; margin-left:50px;"
      >
        <el-form-item label="训练轮数">
          <!-- <el-input v-model="form.name"></el-input> -->
          <el-input v-model="input" :placeholder="dialogShowText.epoch" :disabled="true" />
        </el-form-item>
        <el-form-item label="批量大小">
          <!-- <el-input v-model="form.name"></el-input> -->
          <el-input v-model="input" :placeholder="dialogShowText.batchsize" :disabled="true" />
        </el-form-item>
        <el-form-item label="学习率">
          <el-input v-model="input" :placeholder="dialogShowText.lr" :disabled="true" />
        </el-form-item>
        <el-form-item label="学习率预热(%)">
          <!-- <el-input v-model="form.name"></el-input> -->
          <el-input v-model="input" :placeholder="dialogShowText.lrwarmup" :disabled="true" />
        </el-form-item>
        <el-form-item label="学习率调度器">
          <el-input v-model="input" :placeholder="dialogShowText.lr_scheduler" :disabled="true" />
        </el-form-item>
        <el-form-item label="优化器">
          <el-input v-model="input" :placeholder="dialogShowText.opt" :disabled="true" />
        </el-form-item>
      </el-form>
    </el-dialog>

    <el-dialog :title="textMap[dialogStatus]" :visible.sync="dialogResVisible">
      <!-- <el-form ref="dataForm" :rules="rules" :model="temp" label-position="left" label-width="70px" style="width: 400px; margin-left:50px;">
      </el-form> -->
      <div style="white-space: pre-wrap;">
        {{ trained_res }}
      </div>
    </el-dialog>

    <el-dialog :title="textMap[dialogStatus]" :visible.sync="dialogTrainingResVisible" width="70%" :before-close="handleClose">
      <!-- <el-form ref="dataForm" :rules="rules" :model="temp" label-position="left" label-width="70px" style="width: 400px; margin-left:50px;">
      </el-form> -->
      <div style="white-space: pre-wrap;">
        {{ training_res }}
      </div>
    </el-dialog>

    <el-dialog :visible.sync="dialogPvVisible" title="Reading statistics">
      <el-table :data="pvData" border fit highlight-current-row style="width: 100%">
        <el-table-column prop="key" label="Channel" />
        <el-table-column prop="pv" label="Pv" />
      </el-table>
      <span slot="footer" class="dialog-footer">
        <el-button type="primary" @click="dialogPvVisible = false">{{ $t('table.confirm') }}</el-button>
      </span>
    </el-dialog>
  </div>
</template>

<script>
import { fetchPv, createArticle, updateArticle } from '@/api/article' // fetchList,
import waves from '@/directive/waves' // waves directive
import { parseTime } from '@/utils'
import Pagination from '@/components/Pagination' // secondary package based on el-pagination
import loadFile from '../../utils/index.js'
const calendarTypeOptions = [
  { key: 'CN', display_name: 'China' },
  { key: 'US', display_name: 'USA' },
  { key: 'JP', display_name: 'Japan' },
  { key: 'EU', display_name: 'Eurozone' }
]

// arr to obj, such as { CN : "China", US : "USA" }
const calendarTypeKeyValue = calendarTypeOptions.reduce((acc, cur) => {
  acc[cur.key] = cur.display_name
  return acc
}, {})

export default {
  name: 'ComplexTable',
  components: { Pagination },
  directives: { waves },
  filters: {
    statusFilter(status) {
      const statusMap = {
        Finished: 'success',
        None: 'info',
        Error: 'danger'
      }
      return statusMap[status]
    },
    typeFilter(type) {
      return calendarTypeKeyValue[type]
    }
  },

  data() {
    return {
      timer: 0,
      tableKey: 0,
      list: null,
      total: 0,
      listLoading: true,
      listQuery: {
        page: 1,
        limit: 20,
        importance: undefined,
        title: undefined,
        type: undefined,
        sort: '+id'
      },
      importanceOptions: [1, 2, 3],
      calendarTypeOptions,
      sortOptions: [{ label: 'ID Ascending', key: '+id' }, { label: 'ID Descending', key: '-id' }],
      statusOptions: ['Finished', 'Training', 'deleted'],
      showReviewer: false,
      temp: {
        id: undefined,
        importance: 1,
        remark: '',
        timestamp: new Date(),
        title: '',
        type: '',
        status: 'published'
      },
      dialogFormVisible: false,
      dialogResVisible: false,
      dialogTrainingResVisible: false,
      dialogStatus: '',
      textMap: {
        update: 'Edit',
        create: 'Create',
        show: '模型参数',
        showRes: '训练过程'
      },
      dialogShowText: {
        epoch: '',
        batchsize: '',
        lr: '',
        lrwarmup: '',
        lr_scheduler: '',
        opt: ''
      },
      dialogPvVisible: false,
      pvData: [],
      rules: {
        type: [{ required: true, message: 'type is required', trigger: 'change' }],
        timestamp: [{ type: 'date', required: true, message: 'timestamp is required', trigger: 'change' }],
        title: [{ required: true, message: 'title is required', trigger: 'blur' }]
      },
      downloadLoading: false,
      trained_res: '',
      training_res: ''
    }
  },
  created() {
    this.getList()
    var ModelInfo = this.$route.query
    console.log('SaveSet--', ModelInfo)
    if (ModelInfo.fromSet) {
      const newData = {
        id: this.total + 1,
        InsName: ModelInfo.title,
        createdTime: ModelInfo.summitTime,
        model: ModelInfo.model,
        dataset: ModelInfo.dataset,
        operator: 'LDP',
        status: 'Training',
        modelSet: {
          epoch: ModelInfo.epoch,
          batchsize: ModelInfo.batchsize,
          lr: ModelInfo.lr,
          lrwarmup: ModelInfo.lrwarmup,
          lr_scheduler: ModelInfo.lr_scheduler,
          opt: ModelInfo.opt
        }
      }
      this.list.push(newData)
    }
    // const res = loadFile('./train_log.txt')
    // console.log('文件读取')
    // console.log('res---', res)
  },
  methods: {
    getList() {
      // this.$store.dispatch('user/getModelInfo').then((res) => {
      //   console.log(res)
      //   this.list = res
      //   this.total = this.list.length
      //   this.listLoading = false
      // }).catch(res => {
      //   console.log(res)
      // })
      this.listLoading = false
      this.list = [
        {
          id: 1,
          InsName: 'test1',
          createdTime: '2024-4-09 04:32:27',
          model: 'LayoutLMv3',
          dataset: 'Dataset: 1',
          operator: 'LDP',
          status: 'Finished',
          modelSet: {
            epoch: 20,
            batchsize: 16,
            lr: '1e-5',
            lrwarmup: 10,
            lr_scheduler: 'StepLR',
            opt: 'Adam'
          }
        },
        {
          id: 2,
          InsName: 'test2',
          createdTime: '2024-06-02 23:32:27',
          model: 'LayoutLMv3',
          dataset: 'Dataset: 1',
          operator: 'LDP',
          status: 'Training',
          modelSet: {
            epoch: 20,
            batchsize: 16,
            lr: '1e-5',
            lrwarmup: 10,
            lr_scheduler: 'StepLR',
            opt: 'Adam'
          }
        }]
      this.total = this.list.length
    },
    handleFilter() {
      this.listQuery.page = 1
      this.getList()
    },
    handleModifyStatus(row, status) {
      this.$message({
        message: '操作成功',
        type: 'success'
      })
      row.status = status
    },
    sortChange(data) {
      const { prop, order } = data
      if (prop === 'id') {
        this.sortByID(order)
      }
    },
    sortByID(order) {
      if (order === 'ascending') {
        this.listQuery.sort = '+id'
      } else {
        this.listQuery.sort = '-id'
      }
      this.handleFilter()
    },
    resetTemp() {
      this.temp = {
        id: undefined,
        importance: 1,
        remark: '',
        timestamp: new Date(),
        title: '',
        status: 'published',
        type: ''
      }
    },
    handleShow(row, rowID) {
      this.resetTemp()
      this.dialogStatus = 'show'
      this.dialogFormVisible = true
      console.log(row.modelSet)
      // console.log(rowID)
      this.dialogShowText.epoch = row.modelSet.epoch
      this.dialogShowText.batchsize = row.modelSet.batchsize
      this.dialogShowText.lr = row.modelSet.lr
      this.dialogShowText.lrwarmup = row.modelSet.lrwarmup
      this.dialogShowText.lr_scheduler = row.modelSet.lr_scheduler
      this.dialogShowText.opt = row.modelSet.opt
      console.log(this.dialogShowText)
    },
    handleShowRes(row) {
      this.resetTemp()
      this.dialogStatus = 'showRes'
      // const _this = this
      if (row.status === 'Training') {
        this.dialogTrainingResVisible = true
        this.$store.dispatch('user/getFILE', 'TEMP_LOG.log').then((res) => {
          console.log(res.body)
          this.training_res = res.body
        }).catch(res => {
          console.log(res)
        })

        // setInterval(function() {
        //   _this.$store.dispatch('user/getFILE', 'TEMP_LOG.log').then((res) => {
        //     console.log(res.body)
        //     _this.training_res = res.body
        //   }).catch(res => {
        //     console.log(res)
        //   })

        // }, 1000);
        // let _this = this
        // window.setInterval(_this.$store.dispatch('user/getFILE', 'TEMP_LOG.log').then((res) => {
        //   console.log(res.body)
        //   _this.training_res = res.body
        // }), 2000);
        // this.$store.dispatch('user/getFILE', 'TEMP_LOG.log').then((res) => {
        //   console.log(res.body)
        //   this.training_res = res.body
        // }).catch(res => {
        //   console.log(res)
        // })
      } else if (row.status === 'Finished') {
        this.dialogResVisible = true
        const res = loadFile('./train_log.txt')
        console.log('文件查看')
        this.trained_res = res
        console.log('trained_res---', this.trained_res)
      }
    },
    handleCreate() {
      this.resetTemp()
      this.dialogStatus = 'create'
      this.dialogFormVisible = true
      this.$nextTick(() => {
        this.$refs['dataForm'].clearValidate()
      })
    },
    createData() {
      this.$refs['dataForm'].validate((valid) => {
        if (valid) {
          this.temp.id = parseInt(Math.random() * 100) + 1024 // mock a id
          this.temp.author = 'vue-element-admin'
          createArticle(this.temp).then(() => {
            this.list.unshift(this.temp)
            this.dialogFormVisible = false
            this.$notify({
              title: '成功',
              message: '创建成功',
              type: 'success',
              duration: 2000
            })
          })
        }
      })
    },
    handleUpdate(row) {
      this.temp = Object.assign({}, row) // copy obj
      this.temp.timestamp = new Date(this.temp.timestamp)
      this.dialogStatus = 'update'
      this.dialogFormVisible = true
      this.$nextTick(() => {
        this.$refs['dataForm'].clearValidate()
      })
    },
    updateData() {
      this.$refs['dataForm'].validate((valid) => {
        if (valid) {
          const tempData = Object.assign({}, this.temp)
          tempData.timestamp = +new Date(tempData.timestamp) // change Thu Nov 30 2017 16:41:05 GMT+0800 (CST) to 1512031311464
          updateArticle(tempData).then(() => {
            const index = this.list.findIndex(v => v.id === this.temp.id)
            this.list.splice(index, 1, this.temp)
            this.dialogFormVisible = false
            this.$notify({
              title: '成功',
              message: '更新成功',
              type: 'success',
              duration: 2000
            })
          })
        }
      })
    },
    handleDelete(row, index) {
      this.$notify({
        title: '成功',
        message: '删除成功',
        type: 'success',
        duration: 2000
      })
      this.list.splice(index, 1)
      console.log(index)
    },
    handleFetchPv(pv) {
      fetchPv(pv).then(response => {
        this.pvData = response.data.pvData
        this.dialogPvVisible = true
      })
    },
    handleDownload() {
      this.downloadLoading = true
      import('@/vendor/Export2Excel').then(excel => {
        const tHeader = ['timestamp', 'title', 'type', 'importance', 'status']
        const filterVal = ['timestamp', 'title', 'type', 'importance', 'status']
        const data = this.formatJson(filterVal)
        excel.export_json_to_excel({
          header: tHeader,
          data,
          filename: 'table-list'
        })
        this.downloadLoading = false
      })
    },
    formatJson(filterVal) {
      return this.list.map(v => filterVal.map(j => {
        if (j === 'timestamp') {
          return parseTime(v[j])
        } else {
          return v[j]
        }
      }))
    },
    getSortClass: function(key) {
      const sort = this.listQuery.sort
      return sort === `+${key}` ? 'ascending' : 'descending'
    },
    handleClose(done) {
      this.dialogTrainingResVisible = false
    }
  }
}
</script>
